import unittest,time
from selenium import webdriver
from Page import Page
from Page2 import Page2

class TestCase(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Edge()
        self.driver.maximize_window()


    def tearDown(self):
        time.sleep(3)
        self.driver.quit()

    def test_01(self):
        po = Page(self.driver)
        po.luoji_01()

    def test_02(self):
        po = Page2(self.driver)
        po.luoji_01()
        po.weite()
        time.sleep(1)
        po.person()
        time.sleep(1)
        po.title()
        time.sleep(1)
        po.kj2()
        po.body()
        po.iframe_content()
        po.send()

if __name__ == '__main__':
    unittest.main()